tree data structure in python